home *** CD-ROM | disk | FTP | other *** search
/ Chip 2007 January, February, March & April / Chip-Cover-CD-2007-02.iso / Pakiet bezpieczenstwa / mini Pentoo LiveCD 2006.1 / mpentoo-2006.1.iso / livecd.squashfs / usr / lib / python2.4 / getopt.pyc (.txt) < prev    next >
Python Compiled Bytecode  |  2005-10-18  |  7KB  |  217 lines
  1. # Source Generated with Decompyle++
  2. # File: in.pyc (Python 2.4)
  3.  
  4. """Parser for command line options.
  5.  
  6. This module helps scripts to parse the command line arguments in
  7. sys.argv.  It supports the same conventions as the Unix getopt()
  8. function (including the special meanings of arguments of the form `-'
  9. and `--').  Long options similar to those supported by GNU software
  10. may be used as well via an optional third argument.  This module
  11. provides two functions and an exception:
  12.  
  13. getopt() -- Parse command line options
  14. gnu_getopt() -- Like getopt(), but allow option and non-option arguments
  15. to be intermixed.
  16. GetoptError -- exception (class) raised with 'opt' attribute, which is the
  17. option involved with the exception.
  18. """
  19. __all__ = [
  20.     'GetoptError',
  21.     'error',
  22.     'getopt',
  23.     'gnu_getopt']
  24. import os
  25.  
  26. class GetoptError(Exception):
  27.     opt = ''
  28.     msg = ''
  29.     
  30.     def __init__(self, msg, opt = ''):
  31.         self.msg = msg
  32.         self.opt = opt
  33.         Exception.__init__(self, msg, opt)
  34.  
  35.     
  36.     def __str__(self):
  37.         return self.msg
  38.  
  39.  
  40. error = GetoptError
  41.  
  42. def getopt(args, shortopts, longopts = []):
  43.     '''getopt(args, options[, long_options]) -> opts, args
  44.  
  45.     Parses command line options and parameter list.  args is the
  46.     argument list to be parsed, without the leading reference to the
  47.     running program.  Typically, this means "sys.argv[1:]".  shortopts
  48.     is the string of option letters that the script wants to
  49.     recognize, with options that require an argument followed by a
  50.     colon (i.e., the same format that Unix getopt() uses).  If
  51.     specified, longopts is a list of strings with the names of the
  52.     long options which should be supported.  The leading \'--\'
  53.     characters should not be included in the option name.  Options
  54.     which require an argument should be followed by an equal sign
  55.     (\'=\').
  56.  
  57.     The return value consists of two elements: the first is a list of
  58.     (option, value) pairs; the second is the list of program arguments
  59.     left after the option list was stripped (this is a trailing slice
  60.     of the first argument).  Each option-and-value pair returned has
  61.     the option as its first element, prefixed with a hyphen (e.g.,
  62.     \'-x\'), and the option argument as its second element, or an empty
  63.     string if the option has no argument.  The options occur in the
  64.     list in the same order in which they were found, thus allowing
  65.     multiple occurrences.  Long and short options may be mixed.
  66.  
  67.     '''
  68.     opts = []
  69.     if type(longopts) == type(''):
  70.         longopts = [
  71.             longopts]
  72.     else:
  73.         longopts = list(longopts)
  74.     while args and args[0].startswith('-') and args[0] != '-':
  75.         if args[0] == '--':
  76.             args = args[1:]
  77.             break
  78.         
  79.         if args[0].startswith('--'):
  80.             (opts, args) = do_longs(opts, args[0][2:], longopts, args[1:])
  81.             continue
  82.         (opts, args) = do_shorts(opts, args[0][1:], shortopts, args[1:])
  83.     return (opts, args)
  84.  
  85.  
  86. def gnu_getopt(args, shortopts, longopts = []):
  87.     """getopt(args, options[, long_options]) -> opts, args
  88.  
  89.     This function works like getopt(), except that GNU style scanning
  90.     mode is used by default. This means that option and non-option
  91.     arguments may be intermixed. The getopt() function stops
  92.     processing options as soon as a non-option argument is
  93.     encountered.
  94.  
  95.     If the first character of the option string is `+', or if the
  96.     environment variable POSIXLY_CORRECT is set, then option
  97.     processing stops as soon as a non-option argument is encountered.
  98.  
  99.     """
  100.     opts = []
  101.     prog_args = []
  102.     if isinstance(longopts, str):
  103.         longopts = [
  104.             longopts]
  105.     else:
  106.         longopts = list(longopts)
  107.     if shortopts.startswith('+'):
  108.         shortopts = shortopts[1:]
  109.         all_options_first = True
  110.     elif os.environ.get('POSIXLY_CORRECT'):
  111.         all_options_first = True
  112.     else:
  113.         all_options_first = False
  114.     while args:
  115.         if args[0] == '--':
  116.             prog_args += args[1:]
  117.             break
  118.         
  119.         if args[0][:2] == '--':
  120.             (opts, args) = do_longs(opts, args[0][2:], longopts, args[1:])
  121.             continue
  122.         if args[0][:1] == '-':
  123.             (opts, args) = do_shorts(opts, args[0][1:], shortopts, args[1:])
  124.             continue
  125.         if all_options_first:
  126.             prog_args += args
  127.             break
  128.             continue
  129.         prog_args.append(args[0])
  130.         args = args[1:]
  131.     return (opts, prog_args)
  132.  
  133.  
  134. def do_longs(opts, opt, longopts, args):
  135.     
  136.     try:
  137.         i = opt.index('=')
  138.     except ValueError:
  139.         optarg = None
  140.  
  141.     opt = opt[:i]
  142.     optarg = opt[i + 1:]
  143.     (has_arg, opt) = long_has_args(opt, longopts)
  144.     if has_arg:
  145.         if optarg is None:
  146.             if not args:
  147.                 raise GetoptError('option --%s requires argument' % opt, opt)
  148.             
  149.             optarg = args[0]
  150.             args = args[1:]
  151.         
  152.     elif optarg:
  153.         raise GetoptError('option --%s must not have an argument' % opt, opt)
  154.     
  155.     if not optarg:
  156.         pass
  157.     opts.append(('--' + opt, ''))
  158.     return (opts, args)
  159.  
  160.  
  161. def long_has_args(opt, longopts):
  162.     possibilities = _[1]
  163.     if opt in possibilities:
  164.         return (False, opt)
  165.     elif opt + '=' in possibilities:
  166.         return (True, opt)
  167.     
  168.     if len(possibilities) > 1:
  169.         raise GetoptError('option --%s not a unique prefix' % opt, opt)
  170.     
  171.     if not len(possibilities) == 1:
  172.         raise AssertionError
  173.     unique_match = possibilities[0]
  174.     has_arg = unique_match.endswith('=')
  175.     if has_arg:
  176.         unique_match = unique_match[:-1]
  177.     
  178.     return (has_arg, unique_match)
  179.  
  180.  
  181. def do_shorts(opts, optstring, shortopts, args):
  182.     while optstring != '':
  183.         opt = optstring[0]
  184.         optstring = optstring[1:]
  185.         if short_has_arg(opt, shortopts):
  186.             if optstring == '':
  187.                 if not args:
  188.                     raise GetoptError('option -%s requires argument' % opt, opt)
  189.                 
  190.                 optstring = args[0]
  191.                 args = args[1:]
  192.             
  193.             optarg = optstring
  194.             optstring = ''
  195.         else:
  196.             optarg = ''
  197.         opts.append(('-' + opt, optarg))
  198.     return (opts, args)
  199.  
  200.  
  201. def short_has_arg(opt, shortopts):
  202.     for i in range(len(shortopts)):
  203.         if shortopts[i] == shortopts[i]:
  204.             pass
  205.         elif shortopts[i] != ':':
  206.             return shortopts.startswith(':', i + 1)
  207.             continue
  208.     
  209.     raise GetoptError('option -%s not recognized' % opt, opt)
  210.  
  211. if __name__ == '__main__':
  212.     import sys
  213.     print getopt(sys.argv[1:], 'a:b', [
  214.         'alpha=',
  215.         'beta'])
  216.  
  217.